Skip to content

Conversation

ev-br
Copy link
Member

@ev-br ev-br commented May 8, 2025

closes #311

A matching test is at data-apis/array-api-tests#372

I checked locally that the fix seems to work, but still the sequence is to merge data-apis/array-api-tests#372 and rerun the CI here.

@ev-br ev-br added this to the 1.12 milestone May 8, 2025
@ev-br ev-br closed this May 9, 2025
@ev-br ev-br reopened this May 9, 2025
@ev-br
Copy link
Member Author

ev-br commented May 9, 2025

Ran locally

$ ARRAY_API_TESTS_SKIP_DTYPES=uint16,uint32,uint64 ARRAY_API_TESTS_MODULE=array_api_compat.torch pytest array_api_tests/test_searching_functions.py::test_count_nonzero --max-examples=100_000 

(yes, $10^5$ examples). All seems to be in order.

@ev-br ev-br force-pushed the count_nonzero_torch_tuple_axis_ branch from 60f9310 to 8c62443 Compare May 11, 2025 09:36
@ev-br
Copy link
Member Author

ev-br commented May 11, 2025

CI is green despite numpy 1.22 nonsense; real testing has been done locally (#318 (comment)), including the matching test from array-api-tests. Let's land this.

@ev-br ev-br merged commit 0a14d6c into data-apis:main May 11, 2025
23 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

torch.count_nonzero fails with axis tuple and keepdims=True
1 participant